package com.stephen.data.sort;

import java.util.Arrays;

/**
 * @author: stephen
 * @create: 2020-01-31 19:43
 * @description: 快速排序,800w数据排序,只需要2秒左右
 *  8000w数据,17秒
 */
public class QuickSort {

    public static void main(String[] args) {
        int[] arr = {-9,78,0,23,-567,70,-2,100,20};
        sort(arr,0,arr.length - 1);
        System.out.println(Arrays.toString(arr));

        // 测试排序时间
        int[] arrTest = new int[80000000];
        for (int i = 0; i < 80000000; i++) {
            arrTest[i] = (int)(Math.random() * 8000000);
        }
        long start = System.currentTimeMillis();
        sort(arrTest,0,arrTest.length - 1); // 9743 ms
        long end = System.currentTimeMillis();
        System.out.printf("排序时长: [%d]毫秒",end - start); //[10043]毫秒
        // System.out.println(Arrays.toString(arrTest));

    }

    /**
     * 快速排序算法实现
     * @param arr 数组
     * @param left 数组左下标
     * @param right 数组右下标
     */
    public static void sort(int[] arr,int left,int right) {
        int l = left;
        int r = right;
        int pivot = arr[(left + right) / 2]; // 中轴值
        int temp = 0;

        // 让比 pivot 小的放到左边,比pivot大的放到右边
        while (l < r) {
            // 在pivot的左边一直找,找到大于pivot的值才退出
            while (arr[l] < pivot) {
                l += 1;
            }
            // 在pivot的右边一直找,找到小于等于pivot的值才退出
            while (arr[r] > pivot) {
                r -= 1;
            }
            // l >= r 说明pivot左边全是比pivot小的值
            // 右边全是比pivot大的值
            if (l >= r) {
                break;
            }
            // 交换
            temp = arr[l];
            arr[l] = arr[r];
            arr[r] = temp;

            // 如果交换完后,发现arr[l] == pivot,让r--,前移
            if (arr[l] == pivot) {
                r -= 1;
            }
            // 如果交换完后,发现arr[r] == pivot,让 l++,后移
            if (arr[r] == pivot) {
                l += 1;
            }

        }

        // 不错开的话,会出现死循环
        if (l == r) {
            l += 1;
            r -= 1;
        }

        // 向左递归
        if (left < r) {
            sort(arr,left,r);
        }

        // 向右递归
        if (right > l) {
            sort(arr,l,right);
        }
    }
}
